import os
import sys

# 添加项目根目录到Python路径，以便导入config_reader模块
current_dir = os.path.dirname(os.path.abspath(__file__))
project_root = os.path.dirname(os.path.dirname(current_dir))
sys.path.append(project_root)

from config_reader import get_deepseek_api_key
from openai import OpenAI

class CustomerServiceBot:
    def __init__(self):
        # 初始化DeepSeek客户端
        api_key = get_deepseek_api_key()
        self.client = OpenAI(api_key=api_key, base_url="https://api.deepseek.com")
        
        # 初始化System Prompt，定义客服角色与回答规则
        self.context = [{
            "role": "system",
            "content": """
            你是手机品牌客服，需：
            1. 识别用户问题类型（售后/功能咨询/投诉）
            2. 分步骤解答（每个步骤清晰易懂）
            3. 结尾提供官方联系方式
            """
        }]

    def respond(self, query):
        # 追加用户查询到上下文
        self.context.append({"role": "user", "content": query})

        # 调用DeepSeek模型，保留最近4轮对话（8条消息），temperature=0.3确保严谨
        response = self.client.chat.completions.create(
            model="deepseek-chat",
            messages=self.context[-8:],
            temperature=0.3
        )

        # 追加助手回复到上下文，返回结果
        reply = response.choices[0].message.content
        self.context.append({"role": "assistant", "content": reply})
        return reply


# 实例化Bot并测试
bot = CustomerServiceBot()
print(bot.respond("手机充不进电怎么办？"))